import java.awt.*;
import javax.swing.*;
import java.io.*;
import javax.sound.midi.*;
import java.util.*;
import java.awt.event.*;
import java.net.*;
import javax.swing.event.*;


public class BeatBoxFinal implements MetaEventListener {

      
      JFrame theFrame;
      JPanel mainPanel;
      JList incomingList;
      JTextField userMessage;
      ArrayList checkboxList;
      int bpm = 120;
      int nextNum;
      Vector listVector = new Vector();
      String userName;
      ObjectOutputStream out;
      ObjectInputStream in;
      
      HashMap otherSeqsMap = new HashMap();
      
      
      Sequencer sequencer;
      Sequence sequence;
      Sequence mySequence = null;

      Track track;
      
      String[] instrumentNames = {"Bass Drum", "Closed Hi-Hat", "Open Hi-Hat","Acoustic Snare", "Crash Cymbal", "Hand Clap", "High Tom", "Hi Bongo", "Maracas", "Whistle", "Low Conga", "Cowbell", "Vibraslap", "Low-mid Tom", "High Agogo", "Open Hi Conga"};
      int[] instruments = {35,42,46,38,49,39,50,60,70,72,64,56,58,47,67,63};
    


      public static void main (String[] args) {
         new BeatBoxFinal().startUp(args[0]);  // args[0] is your user ID/screen name
      }

      public void startUp(String name) {
          userName = name;
          // open connection to the server
          try {
            Socket sock = new Socket("127.0.0.1", 4242);
            out = new ObjectOutputStream(sock.getOutputStream());
            in = new ObjectInputStream(sock.getInputStream());
            Thread remote = new Thread(new RemoteReader());
           remote.start();
          } catch(Exception ex) {
               
               System.out.println("couldn't connect - you'll have to play alone.");
          }
         
          setUpMidi();
          buildGUI();
      }

       public void buildGUI() {
                  
          theFrame = new JFrame("Cyber BeatBox");
          BorderLayout layout = new BorderLayout();
          JPanel background = new JPanel(layout);
          background.setBorder(BorderFactory.createEmptyBorder(10,10,10,10));

          checkboxList = new ArrayList();

          Box buttonBox = new Box(BoxLayout.Y_AXIS);
          JButton start = new JButton("Start");
          start.addActionListener(new MyStartListener());
          buttonBox.add(start);
          
          
          JButton stop = new JButton("Stop");
          stop.addActionListener(new MyStopListener());
          buttonBox.add(stop);

          JButton upTempo = new JButton("Tempo Up");
          upTempo.addActionListener(new MyUpTempoListener());
          buttonBox.add(upTempo);

           JButton downTempo = new JButton("Tempo Down");
          downTempo.addActionListener(new MyDownTempoListener());
          buttonBox.add(downTempo);

          JButton sendIt = new JButton("sendIt");
          sendIt.addActionListener(new MySendListener());
          buttonBox.add(sendIt);

          userMessage = new JTextField();
          buttonBox.add(userMessage);

          incomingList = new JList();
         
          incomingList.addListSelectionListener(new MyListSelectionListener());
          incomingList.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);
          JScrollPane theList = new JScrollPane(incomingList);
          buttonBox.add(theList);
         
          incomingList.setListData(listVector); // no data to start with


          Box nameBox = new Box(BoxLayout.Y_AXIS);
          for (int i = 0; i < 16; i++) {
              nameBox.add(new Label(instrumentNames[i]));
          }
        
          background.add(BorderLayout.EAST, buttonBox);
          background.add(BorderLayout.WEST, nameBox);

          theFrame.getContentPane().add(background);
          
          GridLayout grid = new GridLayout(16,16);
          grid.setVgap(1);
          grid.setHgap(2);
          mainPanel = new JPanel(grid);
          background.add(BorderLayout.CENTER, mainPanel);


          for (int i = 0; i < 256; i++) {                    
                JCheckBox c = new JCheckBox();
              
                c.setSelected(false);
                checkboxList.add(c);
                mainPanel.add(c);            
          } // end loop

          theFrame.setBounds(50,50,300,300);
          theFrame.pack();
          theFrame.setVisible(true);
        } // close method


     public void setUpMidi() {
       try {
        sequencer = MidiSystem.getSequencer();
        sequencer.open();
        sequencer.addMetaEventListener(this);
        sequence = new Sequence(Sequence.PPQ,4);
        track = sequence.createTrack();
        sequencer.setTempoInBPM(bpm);
        
       } catch(Exception e) {e.printStackTrace();}
    } // close method



     public void buildTrackAndStart() {
        ArrayList trackList = null; // this will hold the instruments for each 
        // vertical column -- in other words, each tick (may have multiple instruments)
     
        sequence.deleteTrack(track);
        track = sequence.createTrack();

        // now loop through and make an event on and off for each thing that's checked

      for (int i = 0; i < 16; i++) {
         trackList = new ArrayList();

         for (int j = 0; j < 16; j++) {
            JCheckBox jc = (JCheckBox) checkboxList.get(j + (16*i));
            if (jc.isSelected()) { 
               int key = instruments[i];  
               trackList.add(new Integer(key));
            } else {
               trackList.add(null);  // because this slot should be empty in the track
            }
           
        } // close inner

       makeTracks(trackList);
     } // close outer

     track.add(makeEvent(192,9,1,0,15)); // - so we always go to full 16 beats 
                       
       try {
           
           sequencer.setSequence(sequence);                    
           sequencer.start();
           sequencer.setTempoInBPM(bpm);
       } catch(Exception e) {e.printStackTrace();}
 
      } // close method
            
 
      public class MyStartListener implements ActionListener {
        public void actionPerformed(ActionEvent a) {
           
           buildTrackAndStart();
         }
      }

    public class MyStopListener implements ActionListener {
       public void actionPerformed(ActionEvent a) {
           sequencer.stop();
       }
    }

    public class MyUpTempoListener implements ActionListener {
       public void actionPerformed(ActionEvent a) {
           bpm += 3;
          
        }
     }

     public class MyDownTempoListener implements ActionListener {
         public void actionPerformed(ActionEvent a) {
            bpm -= 3;
        }
    }

    public class MySendListener implements ActionListener {

       public void actionPerformed(ActionEvent a) {
          // make an arraylist of just the STATE of the checkboxes
         boolean[] checkboxState = new boolean[256];

         for (int i = 0; i < 256; i++) {
             JCheckBox check = (JCheckBox) checkboxList.get(i);
             if (check.isSelected()) {
                checkboxState[i] = true;
             }
          } // close for
         
         String messageToSend = null;         

         try {
           // FileOutputStream fileStream = new FileOutputStream(new File("Checkbox.ser"));
            //ObjectOutputStream os = new ObjectOutputStream(fileStream);
            out.writeObject(userName + nextNum++ + ": " + userMessage.getText());
            out.writeObject(checkboxState);
         } catch(Exception ex) {
             System.out.println("Sorry dude. Could not send it to the server.");
         }
         userMessage.setText("");         

       } // close method
    } // close inner class


    public class MyListSelectionListener implements ListSelectionListener {

        public void valueChanged(ListSelectionEvent le) {
          if (!le.getValueIsAdjusting()) {
            String selected = (String) incomingList.getSelectedValue();  
            if (selected != null) {    
            
                // now go to the map, and change the sequence
                 boolean[] selectedState = (boolean[]) otherSeqsMap.get(selected);
            
                 changeSequence(selectedState);                                     
            
                 sequencer.stop(); 
                 buildTrackAndStart();
             }
         }
      } // close method
  } // close inner class


   public class RemoteReader implements Runnable {
        boolean[] checkboxState = null;
        String nameToShow = null;
        Object obj = null;
        public void run() {
          try {
              while((obj=in.readObject()) != null) {
              System.out.println("got an object from server");
              System.out.println(obj.getClass());
            
              String nameToShow = (String) obj;
              checkboxState = (boolean[]) in.readObject();
              otherSeqsMap.put(nameToShow, checkboxState);

              listVector.add(nameToShow);
              incomingList.setListData(listVector);
              // now reset the sequence to be this
              } // close while 
            } catch(Exception ex) {ex.printStackTrace();}  
                 
            
        } // close run
   } // close inner class


    public class MyPlayMineListener implements ActionListener {
       public void actionPerformed(ActionEvent a) {
          if (mySequence != null) {
             sequence = mySequence;  // restore to my original
          }
       }
    }

   public void changeSequence(boolean[] checkboxState) {
       
          for (int i = 0; i < 256; i++) {
             JCheckBox check = (JCheckBox) checkboxList.get(i);
             if (checkboxState[i]) {
                check.setSelected(true);
             } else {
                check.setSelected(false);
             }
          } // close for  
     } // close method
    

     public void makeTracks(ArrayList list) {
        
         Iterator it = list.iterator();
         // which key was pressed?
         
         for (int i = 0; i < 16; i++) {
            Integer num = (Integer) it.next();
            if (num != null) {
               int numKey = num.intValue();            
            
               track.add(makeEvent(144,9,numKey, 100, i));
               track.add(makeEvent(128,9,numKey,100, i + 1));
             }
          }
     }
 
     public  MidiEvent makeEvent(int comd, int chan, int one, int two, int tick) {
          MidiEvent event = null;
          try {
            ShortMessage a = new ShortMessage();
            a.setMessage(comd, chan, one, two);
            event = new MidiEvent(a, tick);
            
            }catch(Exception e) { }
          return event;
       }

      public void meta(MetaMessage message) {
            if (message.getType() == 47) {
                sequencer.start();
                sequencer.setTempoInBPM(bpm);
            }
       }

} // close class

        
             
          
          
          